package com.njbw.tool;

import lombok.extern.log4j.Log4j2;
import org.apache.commons.lang3.StringUtils;

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.net.URL;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import java.util.Properties;

/**
 * 动态属性读取
 * 属性文件读取工具类
 *
 * 通过 Spring 的 @Value、@ImportResource 都可以实现属性读取，这里主要是为了讲一个动态配置
 * nacos、Spring Config 都可以实现，那边改，改了之后我这边立即生效
 * 平时需要重启服务才能获得最新的配置
 */
@Log4j2
public class PropertiesUtil {

    /**
     * 属性文件与属性集合的关系
     * 文件和属性集合的一个对应关系
     */
    private Map<String, Properties> propertiesMap = new HashMap<>();

    /**
     * 属性文件和修改时间的关系
     */
    private Map<String, Long> modifyTimeMap = new HashMap<>();

    /**
     * 前缀路径
     * 指定的属性文件的目录路径
     */
    private String configPath = "";

    private void setConfigPath(String configPath) {
        this.configPath = configPath;
    }

    private PropertiesUtil() {
        if (SingleHolder.INSTANCE != null) {
            throw new RuntimeException("不能创建多个对象！");
        }
    }

    private static class SingleHolder {
        private static PropertiesUtil INSTANCE = new PropertiesUtil();
    }

    public static PropertiesUtil getInstance() {
        return SingleHolder.INSTANCE;
    }

    public Object readResolve() {
        return SingleHolder.INSTANCE;
    }

    /**
     * 获取指定属性的值
     */
    public String getPropertyValue(String propertyFileName, String key) {
        String fileName = convertFileName(propertyFileName);
        try {
            if (!propertiesMap.containsKey(fileName)) {
                loadProperties(fileName);
            } else {
                checkPropertiesModified(fileName);
            }
            return propertiesMap.get(fileName).getProperty(key);
        } catch (Exception e) {
            log.error("PropertiesUtils.getPropertyValue.error:{}", e.getMessage(), e);
        }
        return "";
    }

    /**
     * 检查属性文件有无更新，若更新则重新加载
     */
    private void checkPropertiesModified(String fileName) throws Exception {
        // 1.拿到 File 文件
        File file = getPropertiesFile(fileName);
        long newTime = file.lastModified();
        Long lastModifiedTime = modifyTimeMap.get(fileName);
        if (newTime == 0) {
            if (lastModifiedTime == null) {
                log.error(fileName + ".properties file does not exist!");
            }
        } else if (newTime > lastModifiedTime) {
            loadProperties(fileName);
        }
    }

    /**
     * 把配置文件加载进 Map 方便我们进行读取
     * 加载指定名称的属性文件
     */
    private void loadProperties(String fileName) throws Exception {
        // 1.拿到 File 文件
        File file = getPropertiesFile(fileName);
        // 2.拿到更新时间
        Long newTime = file.lastModified();
        if (propertiesMap.get(fileName) != null) {
            propertiesMap.remove(fileName);
        }
        Properties properties = new Properties();
        properties.load(new InputStreamReader(new FileInputStream(file), StandardCharsets.UTF_8));
        propertiesMap.put(fileName, properties);
        modifyTimeMap.put(fileName, newTime);
    }

    /**
     * 获取属性文件的去路径，首先
     */
    private File getPropertiesFile(String fileName) throws Exception {
        File propertiesFile;
        if (StringUtils.isNotBlank(this.configPath)) {
            return new File(this.configPath + File.separator + fileName + ".properties");
        }
        // 从用户目录下获取
        String dir = System.getProperty("user.dir") + File.separator + fileName + ".properties";
        propertiesFile = new File(dir);
        if (!propertiesFile.exists()) {
            // 从项目目录下获取
            URL url = PropertiesUtil.class.getResource("/" + fileName + ".properties");
            if (Objects.nonNull(url)) {
                propertiesFile = new File(url.toURI());
            } else {
                propertiesFile = null;
            }
        }
        return propertiesFile;
    }

    /**
     * 对传入的属性文件名称进行处理，如果包含.properties后缀则去掉
     */
    private String convertFileName(String fileName) {
        if (!fileName.endsWith(".properties")) {
            return fileName;
        }
        int index = fileName.lastIndexOf(".");
        fileName = fileName.substring(0, index);
        return fileName;
    }
}
